#调用接口进行在线转换，1从weights转trt,2从onnx转trt，再通过ssh把转好的模型发送到服务器
#python3 change_model.py 0 1 yolov4 cap 1 608
import os
import sys


device=sys.argv[1] #使用GPU id
model_way=sys.argv[2] #1从weights转trt,2从onnx转trt
model_type=sys.argv[3] #模型类型
model_name=sys.argv[4] #模型名称
model_batch=sys.argv[5] #并发数
model_size=sys.argv[6] #分辨率

change_str="./trtexec --onnx=/home/xs/AI_docker/models/onnx/"+str(model_name)+"/"+str(model_batch)+"/"+str(model_size)+"/model.onnx --explicitBatch  --saveEngine=/home/xs/AI_docker/models/yolov4/"+str(model_name)+"/"+str(model_batch)+"/"+str(model_size)+"/model.trt --fp16"

change_str1="cd /home/xs/TensorRT-7.1.3.4/bin"
print(os.system(change_str1+"&&"+change_str))

